package org.example.message;

import com.alibaba.ttl.threadpool.TtlExecutors;
import com.google.common.base.Throwables;
import com.google.common.util.concurrent.ThreadFactoryBuilder;
import lombok.extern.slf4j.Slf4j;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

import java.util.concurrent.ExecutorService;
import java.util.concurrent.LinkedBlockingQueue;
import java.util.concurrent.ThreadPoolExecutor;
import java.util.concurrent.TimeUnit;

/**
 * @author weigang
 * @create 2022-10-21 16:21
 */

@SuppressWarnings("AlibabaThreadShouldSetName")
@Configuration
@Slf4j
public class MessagelConfig {
    @Bean("messagePool")
    public ExecutorService asyncEmailPool() {
        /*
        * JDK的InheritableThreadLocal类可以完成父子线程值的传递。
        * 但对于使用线程池等会缓存线程的组件的情况，线程由线程池创建好，并且线程是缓存起来反复使用的；
        * 这时父子线程关系的上下文传递已经没有意义，应用中要做上下文传递，
        * 实际上是在把  任务提交给线程池时的上下文传递  到  任务执行  时。
        * TtlExecutors本库提供的TransmittableThreadLocal类继承并加强InheritableThreadLocal类，解决上述的问题，
        *
        * TTL（transmittable-thread-local）是一个线程间传递ThreadLocal，异步执行时上下文传递的解决方案。
        * 整个库的核心是构建在TransmittableThreadLocal类（继承并加强InheritableThreadLocal类）之上，
        * 同时包含线程池修饰（ExecutorService/ForkJoinPool/TimerTask）以及Java Agent支持，代码小于1k行，短小精悍。
        *
        * */
        return TtlExecutors.getTtlExecutorService(
                new ThreadPoolExecutor(1, 4, 30, TimeUnit.SECONDS,
                        new LinkedBlockingQueue<>(200),
                        new ThreadFactoryBuilder()
                                .setNameFormat("message-pool-%d")
                                .setUncaughtExceptionHandler((thread, ex) -> {
                                    log.info("message pool raised error, ex:{}", Throwables.getStackTraceAsString(ex));
                                }).build(),
                        /*执行*/
                        (r, executor) -> {
                            log.info("message discard, thread pool reject");
                        }));
    }
}
